package com.spider.im.common.utils;

import cn.hutool.core.bean.BeanUtil;
import com.spider.im.common.annotation.Desensitize;
import com.spider.im.common.annotation.DesensitizeStrategy;

import java.lang.reflect.Field;

/**
 * 脱敏工具类，用于对对象中的敏感字段进行脱敏处理。
 */
public class DesensitizationUtils {

    /**
     * 对传入的对象或字符串进行脱敏处理。
     *
     * @param obj 待脱敏的对象
     * @return 脱敏后的新对象（原对象会被修改）
     * @throws IllegalAccessException 如果反射访问字段失败
     */
    public static Object desensitizeObject(Object obj) throws IllegalAccessException {

        if (obj == null) {
            return null;
        }
//        // 若输入是字符串类型，使用默认策略（如身份证）进行脱敏
//        if (obj instanceof String) {
//            // 默认策略可以根据需要调整
//            return desensitize((String) obj, DesensitizeStrategy.PHONE_NUMBER);
//        }

        // 获取对象所有声明字段
        Field[] fields = obj.getClass().getDeclaredFields();
        for (Field field : fields) {
            // 检查字段是否标注了 @Desensitize 注解
            if (field.isAnnotationPresent(Desensitize.class)) {
                // 设置字段可访问
                field.setAccessible(true);

                // 获取字段原始值
                String value = (String) field.get(obj);

                // 获取注解中定义的脱敏策略
                DesensitizeStrategy strategy = field.getAnnotation(Desensitize.class).strategy();

                // 执行脱敏操作
                String desensitizedValue = desensitize(value, strategy);

                // 将脱敏后的值设置回字段
                field.set(obj, desensitizedValue);
            }
        }

        return obj;
    }

    /**
     * 根据指定策略对字符串进行脱敏处理。
     *
     * @param value     原始字符串
     * @param strategy  脱敏策略
     * @return 脱敏后的字符串
     */
    private static String desensitize(String value, DesensitizeStrategy strategy) {
        if (value == null) {
            return null;
        }
        switch (strategy) {
            case ID_CARD:
                // 保留前6位和后4位，中间用*代替
                return value.substring(0, 6) + "********" + value.substring(14);
            case PHONE_NUMBER:
                // 保留前3位和后4位，中间用*代替
                return value.substring(0, 3) + "*******" + value.substring(10);
            case EMAIL:
                // 邮箱脱敏：保留首尾字符，中间用*代替
                int atIndex = value.indexOf('@');
                String localPart = value.substring(0, atIndex);
                return localPart.charAt(0)
                        + "*".repeat(localPart.length() - 2)
                        + localPart.charAt(localPart.length() - 1)
                        + value.substring(atIndex);
            default:
                // 默认不脱敏
                return value;
        }
    }

    /**
     * 对传入的对象或字符串进行脱敏处理。
     *
     * @param obj 待脱敏的对象
     * @return 脱敏后的新对象（原对象会被修改）
     * @throws IllegalAccessException 如果反射访问字段失败
     */
    public static Object desensitizeNewObject(Object obj) throws IllegalAccessException {

        if (obj == null) {
            return null;
        }
        Object newObj = obj;
        //拷贝obj的内容到newObj
//        BeanUtil.copyProperties(obj,newObj);
//        // 若输入是字符串类型，使用默认策略（如身份证）进行脱敏
//        if (newObj instanceof String) {
//            // 默认策略可以根据需要调整
//            return desensitize((String) newObj, DesensitizeStrategy.PHONE_NUMBER);
//        }

        // 获取对象所有声明字段
        Field[] fields = newObj.getClass().getDeclaredFields();
        for (Field field : fields) {
            // 检查字段是否标注了 @Desensitize 注解
            if (field.isAnnotationPresent(Desensitize.class)) {
                // 设置字段可访问
                field.setAccessible(true);

                // 获取字段原始值
                String value = (String) field.get(newObj);

                // 获取注解中定义的脱敏策略
                DesensitizeStrategy strategy = field.getAnnotation(Desensitize.class).strategy();

                // 执行脱敏操作
                String desensitizedValue = desensitize(value, strategy);

                // 将脱敏后的值设置回字段
                field.set(newObj, desensitizedValue);
            }
        }

        return newObj;
    }

}
